from selenium import webdriver
from selenium.webdriver.common.by import By
from time import sleep

driver = webdriver.Firefox()
driver.maximize_window()
driver.get('file:///D:/nsd2404/html/register.html')

sleep(3)
# 定位链接中含有A网页的元素，执行点击
driver.find_element(By.PARTIAL_LINK_TEXT, 'A网页').click()
# 获取所有窗口的句柄
handles = driver.window_handles
print('所有窗口：', handles)
# 获取主窗口的句柄
current  = driver.current_window_handle
print('当前窗口：', current)
# 切换窗口：需要找到新打开的窗口的句柄
for item in handles:
    if item != current:  # 如果不是主窗口，就是新打开的窗口
        # 切换窗口
        driver.switch_to.window(item)
        # 往A网页输入内容
        driver.find_element(By.CSS_SELECTOR, '#userA').send_keys('admin')

sleep(3)
driver.quit()